package com.fengye.algorithms.leecode;

import com.fengye.algorithms.leecode.common.TreeNode;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author fengye
 * @Date 2023/4/8 11:14
 **/
public class TreeTraversal94And144ANd145 {
    /**
     * 145. 二叉树的后序遍历
     * @param root
     * @return
     */
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null) {
            return result;
        }
        doPostDfs(root, result);
        return result;
    }
    public void doPostDfs(TreeNode root, List<Integer> result) {
        if(root == null) {
            return;
        }

        if(root.left != null) {
            doPostDfs(root.left, result);
        }
        if(root.right != null) {
            doPostDfs(root.right, result);
        }
        result.add(root.val);
    }

    /**
     * 94. 二叉树的中序遍历
     * @param root
     * @return
     */
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null) {
            return result;
        }
        doInDfs(root, result);
        return result;
    }

    public void doInDfs(TreeNode node, List<Integer> result) {
        if(node == null) {
            return;
        }

        if(node.left!=null) {
            doInDfs(node.left, result);
            result.add(node.val);
        }

        if(node.right!=null) {
            doInDfs(node.right, result);
            result.add(node.val);
        }

    }

    /**
     * 144. 二叉树的前序遍历
     * @param root
     * @return
     */
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null) {
            return result;
        }

        doDfs(root, result);
        return result;
    }

    public void doDfs(TreeNode node, List<Integer> result) {
        if(node == null) {
            return;
        }
        result.add(node.val);
        if(node.left != null) {
            doDfs(node.left,result);
        }

        if(node.right != null) {
            doDfs(node.right, result);
        }
    }
}
